"""
创建一个简单的测试图像用于keypoint extraction测试
"""

import cv2
import numpy as np
from pathlib import Path


def create_test_image(output_path='input_images/test.jpg'):
    """创建一个简单的测试图像"""
    # 创建800x600的图像
    h, w = 600, 800
    img = np.ones((h, w, 3), dtype=np.uint8) * 255  # 白色背景
    
    # 绘制一些彩色矩形作为前景物体
    cv2.rectangle(img, (100, 100), (300, 250), (0, 255, 0), -1)
    cv2.rectangle(img, (400, 150), (600, 300), (255, 0, 0), -1)
    cv2.rectangle(img, (150, 350), (350, 500), (0, 0, 255), -1)
    cv2.rectangle(img, (450, 350), (650, 500), (255, 255, 0), -1)
    
    # 绘制一些圆形
    cv2.circle(img, (200, 400), 50, (255, 0, 255), -1)
    cv2.circle(img, (550, 200), 60, (0, 255, 255), -1)
    
    # 保存图像
    output_path = Path(output_path)
    output_path.parent.mkdir(exist_ok=True, parents=True)
    cv2.imwrite(str(output_path), img)
    
    print(f"测试图像已创建: {output_path}")
    return img


if __name__ == '__main__':
    create_test_image()

